Check if divider not zeroΒΆ
def check_div(func):
def check(a, b):
if b == 0:
print("Cant't divide by zero")
return
return func(a, b)
return check
@check_div
def divide(a, b):
return a // b
# divide = check_div(divide)
print(divide(10, 2))
print(divide(2, 0))
print(divide(15, 5))